iT邦幫忙

2018 iT 邦幫忙鐵人賽
DAY 9
0
AI & Machine Learning

深度學習環境安裝筆記系列 第 9

[Day 09] 深度學習環境安裝筆記 — 編譯與安裝 Caffe

  • 分享至 

  • xImage
  •  

Caffe 其實蠻好編譯的,只是有幾個地方要注意一下,那廢話不多說,就讓我們開始吧

編譯 Caffe

1.安裝編譯 Caffe 需要的套件

sudo apt install libboost1.58-all-dev libprotobuf-dev libleveldb-dev libsnappy-dev libhdf5-serial-dev protobuf-compiler libatlas-base-dev libgflags-dev libgoogle-glog-dev liblmdb-dev python-pip python-scipy python3-scipy -y

2. 下載 Caffe 的原始碼

git clone https://github.com/BVLC/caffe.git

3. 下載並安裝 Intel MKL

下載前要先註冊會員,這邊沒什麼我就不仔細講解了。

解壓縮並安裝 MKL

tar xvfz l_mkl_2018.1.163.tgz
cd l_mkl_2018.1.163
sudo ./install_GUI.sh

增加 MKL 的環境變數到 .bashrc ,並用 source ~/.bashrc 來更新環境變數

source /opt/intel/mkl/bin/mklvars.sh intel64

4.

5. 修改 Caffe 的配置

  1. 複製設定檔
cd caffe
cp Makefile.config.example ./Makefile.config
vim Makefile.config
  1. 將 USE_CUDNN 取消註解
  2. 如果是使用 OpenCV 3 的要把 OPENCV_VERSION := 3 取消註解
  3. 修改 BLAS := atlas 為 BLAS := mkl
  4. 如果要使用 Python3 的,要註解掉 69,70 行,然後取消註解 79,80,81 行
  5. 增加 HDF5 的引用,修改 95,96 行為下面的樣子
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial  
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/hdf5/serial

7.取消註解 WITH_PYTHON_LAYER

  1. 解決編譯的時候的 HDF5 錯誤
cd /usr/lib/x86_64-linux-gnu/
sudo ln -s libhdf5_serial.so.10.1.0 ./libhdf5.so
sudo ln -s libhdf5_serial_h1.so.10.1.0 ./libhdf5_h1.so
  1. 安裝缺少的 Python 套件
cd caffe
cd python
pip install -U --user setuptools
for req in $(cat requirements.txt); do pip install $req -U --user; done
  1. 使用以下指令編譯
make all -j8
make test -j8
make pycaffe -j8

  1. 加入 pycaffe 環境變數至 .bashrc ,並用 source ~/.bashrc 更新
export PYTHONPATH=$PYTHONPATH:~/caffe/python
  1. 在 python 裡 import caffe ,如果能 import 成功就安裝完成


上一篇
[Day 08] 深度學習環境安裝筆記 — 安裝與編譯 OpenCV 02. OpenCV 3.4
下一篇
[Day 10] 深度學習環境安裝筆記 - 安裝 Tensorflow
系列文
深度學習環境安裝筆記13
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言